#!/bin/bash

srun --ntasks=4 --container-image="${CONT}" --container-mounts="${MOUNTS}" --network sharp bash -cx " \
    cd /dataset/criteo_kaggle/dcn &&
    python3 /workdir/test/pybind_test/dcn_4node_2gpu.py /workdir/test/scripts/dcn_4node_2gpu.json"

srun --ntasks=2 --container-image="${CONT}" --container-mounts="${MOUNTS}" --network sharp bash -cx " \
    cd /dataset/criteo_kaggle/criteo && \
    python3 /workdir/test/pybind_test/criteo_2node_4gpu.py /workdir/test/scripts/criteo_2node_4gpu.json"